import tushare as ts
import pandas as pd

# 设置token
ts.set_token('843b322236a08b237ab3fcbc9196045ee5ff5ed071152b1e8d052ef1')
pro = ts.pro_api()

# 获取贵州茅台（600519.SH）的日线数据
df = pro.daily(
    ts_code='600519.SH',  # 股票代码，格式：代码+市场（SH=沪市，SZ=深市）
    start_date='20230101',  # 开始日期（YYYYMMDD）
    end_date='20231231'  # 结束日期
)

# 数据格式转换（将日期转为标准格式，调整列顺序）
df['trade_date'] = pd.to_datetime(df['trade_date'], format='%Y%m%d')
df = df[['trade_date', 'open', 'high', 'low', 'close', 'vol']]  # 保留常用列
df.columns = ['日期', '开盘价', '最高价', '最低价', '收盘价', '成交量']

# 保存为CSV
df.to_csv('贵州茅台2023行情.csv', index=False)
print(df.head())
